import pymysql

#往学生表新增数据
no = int(input('学号: '))
name = input('姓名: ')
location = input('地址: ')
col_no = int(input("学院id："))
sex=int(input("性别："))
birthday=input("出生日期：")
# 1. 创建连接（Connection）
conn = pymysql.connect(host='127.0.0.1', port=3306,
                       user='root', password='qwe123',
                       database='school', charset='utf8mb4')
try:
    # 2. 获取游标对象（Cursor）
    with conn.cursor() as cursor:
        # 3. 通过游标对象向数据库服务器发出SQL语句
        affected_rows = cursor.execute(
            f"insert into `tb_student` (stu_id,stu_name,stu_sex,stu_birth,stu_addr,col_id) values ({no},'{name}',{sex},'{birthday}','{location}',{col_no})",
        )
        
        if affected_rows == 1:
            print('新增学生成功!!!')
    # 4. 提交事务（transaction）
    conn.commit()
except pymysql.MySQLError as err:
    # 4. 回滚事务
    conn.rollback()
    print(type(err), err)
finally:
    # 5. 关闭连接释放资源
    conn.close()